while loop in swift 4